from PIL import Image  
import numpy as np  
import torch  


def png2tensor(image_path):  
    image = Image.open(image_path)  
    image = image.convert('RGB') 
    image_array = np.array(image)
    image_tensor = torch.from_numpy(image_array.transpose((2, 0, 1))).float()
    return image_tensor[0]

得到了图片的tensor表示，就可以通过深度学习做映射。图像的分类算法。一个是感知机。